#!/usr/bin/env python3
# -*- coding:utf-8 -*-

import pygraphviz as pgv

class GraphWrapper(pgv.AGraph):
    # string must be None for render_subgraph
    def __init__(self, string=None, strict=False, **kwargs):
        super().__init__(string=string, strict=strict, **kwargs)

    def write(self, filename):
        return super().write(filename)

    def draw(self, filename, format=None, prog='dot', args=""):
        return super().draw(filename, format=format, prog=prog, args=args)
